from langchain_core.messages import SystemMessage, HumanMessage
from config.load_key import load_key

# from langchain_openai import ChatOpenAI
# llm = ChatOpenAI(
#     model="Qwen/Qwen3-8B",
#     api_key=load_key("siliconflow_api_key"),
#     base_url=load_key("siliconflow_base_url")
# )
# ai_response = llm.invoke([
#     SystemMessage("帮我将英文翻译成中文"),
#     HumanMessage("Hello, how are you?"),
# ])
# print(ai_response)


# from langchain_qwq import ChatQwen
# llm = ChatQwen(
#     model="Qwen/Qwen3-8B",
#     api_key=load_key("siliconflow_api_key"),
#     base_url=load_key("siliconflow_base_url")
# )
# ai_response = llm.invoke([
#     SystemMessage("帮我将英文翻译成中文"),
#     HumanMessage("Hello, how are you?"),
# ])
# print(ai_response)


from langchain_deepseek import ChatDeepSeek
llm = ChatDeepSeek(
    model="deepseek-reasoner",
    api_key=load_key("deepseek_api_key")
)
ai_response = llm.invoke([
    SystemMessage("帮我将英文翻译成中文"),
    HumanMessage("Hello, how are you?"),
])
print(ai_response)
